package com.nuts.base.utils;

import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

import org.apache.commons.lang3.StringUtils;

import lombok.extern.slf4j.Slf4j;

/**
 * 文件工具类
 *
 * @author 许自辉
 * @date 2017年10月17日
 */
@Slf4j
public class FileUtil {
	/**
	 * 读文件得到byte[]
	 * 
	 * @param filename
	 * @return
	 * @throws IOException
	 */
	public static byte[] getContent(String filename) throws IOException {
		File f = new File(filename);
		if (!f.exists()) {
			throw new FileNotFoundException(filename);
		}
		ByteArrayOutputStream bos = new ByteArrayOutputStream((int) f.length());
		BufferedInputStream in = null;
		try {
			in = new BufferedInputStream(new FileInputStream(f));
			int bufSize = 1024;
			byte[] buffer = new byte[bufSize];
			int len = 0;
			while (-1 != (len = in.read(buffer, 0, bufSize))) {
				bos.write(buffer, 0, len);
			}
			return bos.toByteArray();
		} catch (IOException e) {
			throw e;
		} finally {
			try {
				in.close();
			} catch (IOException e) {
				throw e;
			}
			bos.close();
		}
	}

	/**
	 * 读文件得到byte[]
	 * 
	 * @param file
	 * @return
	 * @throws IOException
	 */
	public static byte[] getContentByFile(File file) throws IOException {
		if (!file.exists()) {
		}
		ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length());
		BufferedInputStream in = null;
		try {
			in = new BufferedInputStream(new FileInputStream(file));
			int bufSize = 1024;
			byte[] buffer = new byte[bufSize];
			int len = 0;
			while (-1 != (len = in.read(buffer, 0, bufSize))) {
				bos.write(buffer, 0, len);
			}
			return bos.toByteArray();
		} catch (IOException e) {
			throw e;
		} finally {
			try {
				in.close();
			} catch (IOException e) {
				throw e;
			}
			bos.close();
		}
	}

	/**
	 * 根据文件路径创建文件
	 * 
	 * @author 许自辉
	 * @date 2017年10月17日
	 * @param destDirName
	 *            文件路径
	 * @return
	 */
	public static boolean createDir(String destDirName) {
		if (destDirName.indexOf(".") > 0) {
			if (destDirName.contains(File.separator)) {
				destDirName = destDirName.substring(0, destDirName.lastIndexOf(File.separator));
			}
		}
		File dir = new File(destDirName);
		// 判断目录是否存在
		if (dir.exists()) {
			return false;
		}
		// 结尾是否以File.separator结束
		if (!destDirName.endsWith(File.separator)) {
			destDirName = destDirName + File.separator;
		}
		// 创建目标目录
		if (dir.mkdirs()) {
			return true;
		} else {
			return false;
		}
	}

	/**
	 * 根据创建文件夹并创建文件
	 * 
	 * @author 许自辉
	 * @date 2017年10月10日
	 * @param filePath
	 *            文件路径
	 * @param dates
	 *            写入的字符流
	 * @return
	 * @throws Exception
	 */
	public static boolean writeFile(String filePath, byte[] dates) throws Exception {
		File file = new File(filePath);
		File fileParent = file.getParentFile();
		// 如果文件夹不存在，则先创建文件夹
		if (!fileParent.exists()) {
			fileParent.mkdirs();
		}
		// 如果文件不存在，则先创建文件夹
		if (!file.exists()) {
			file.createNewFile();
			FileOutputStream fs = null;
			try {
				fs = new FileOutputStream(file);
				fs.write(dates);
			} catch (Exception e) {
				throw e;
			} finally {
				try {
					if (fs != null) {
						fs.close();
					}
				} catch (IOException e) {
					throw e;
				}
			}
		}
		return true;
	}

	/**
	 * 根据创建文件夹并创建文件
	 * 
	 * @author 许自辉
	 * @date 2017年10月10日
	 * @param filePath
	 *            文件路径
	 * @param inputStream
	 *            写入的字符流
	 * @return
	 * @throws Exception
	 */
	public static boolean writeFile(String filePath, InputStream inputStream) throws Exception {
		if (StringUtils.isBlank(filePath) || null == inputStream) {
			return false;
		}
		byte[] data = new byte[1024];
		int len = 0;
		FileOutputStream fileOutputStream = null;
		try {
			fileOutputStream = new FileOutputStream(filePath);
			while ((len = inputStream.read(data)) != -1) {
				fileOutputStream.write(data, 0, len);
			}
		} catch (IOException e) {
			throw e;
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException e) {
					throw e;
				}
			}
			if (fileOutputStream != null) {
				try {
					fileOutputStream.close();
				} catch (IOException e) {
					throw e;
				}
			}

		}
		return true;
	}

	/**
	 * 删除文件，可以是文件或文件夹
	 *
	 * @param file
	 *            要删除的文件名
	 * @return 删除成功返回true，否则返回false
	 */
	public static boolean delete(File file) {
		if (!file.exists()) {
			log.error("删除文件失败:" + file.getName() + "不存在！");
			return false;
		} else {
			if (file.isFile()) {
				return deleteFile(file);
			}
			// else
			// return deleteDirectory(file);
		}
		return false;
	}

	/**
	 * 删除单个文件
	 *
	 * @param file
	 *            要删除的文件的文件名
	 * @return 单个文件删除成功返回true，否则返回false
	 */
	public static boolean deleteFile(File file) {
		// 如果文件路径所对应的文件存在，并且是一个文件，则直接删除
		if (file.exists() && file.isFile()) {
			if (file.delete()) {
				log.error("删除单个文件" + file.getName() + "成功！");
				return true;
			} else {
				log.error("删除单个文件" + file.getName() + "失败！");
				return false;
			}
		} else {
			log.error("删除单个文件失败：" + file.getName() + "不存在！");
			return false;
		}
	}

	/**
	 * 删除目录及目录下的文件
	 *
	 * @param dir
	 *            要删除的目录的文件路径
	 * @return 目录删除成功返回true，否则返回false
	 */
	public static boolean deleteDirectory(String dir) {
		// 如果dir不以文件分隔符结尾，自动添加文件分隔符
		if (!dir.endsWith(File.separator)) {
			dir = dir + File.separator;
		}

		File dirFile = new File(dir);
		// 如果dir对应的文件不存在，或者不是一个目录，则退出
		if ((!dirFile.exists()) || (!dirFile.isDirectory())) {
			log.error("删除目录失败：" + dir + "不存在！");
			return false;
		}
		boolean flag = true;
		// 删除文件夹中的所有文件包括子目录
		File[] files = dirFile.listFiles();
		for (int i = 0; i < files.length; i++) {
			// 删除子文件
			if (files[i].isFile()) {
				flag = deleteFile(files[i]);
				if (!flag) {
					break;
				}
			}
			// 删除子目录
			else if (files[i].isDirectory()) {
				flag = deleteDirectory(files[i].getAbsolutePath());
				if (!flag) {
					break;
				}
			}
		}
		if (!flag) {
			log.error("删除目录失败！");
			return false;
		}
		// 删除当前目录
		if (dirFile.delete()) {
			log.error("删除目录" + dir + "成功！");
			return true;
		} else {
			return false;
		}
	}

}
